

class_id_to_name = {
    "1": "bathtub",
    "2": "bed",
    "3": "chair",
    "4": "desk",
    "5": "dresser",
    "6": "monitor",
    "7": "night_stand",
    "8": "sofa",
    "9": "table",
    "10": "toilet",
          
    "11": "airplane",
    "12": "bench",
    "13": "bookshelf",
    "14": "bottle",
    "15": "bowl",
    "16": "car",
    "17": "cone",
    "18": "cup",
    "19": "curtain",
    "20": "door",

    "21": "flower_pot",
    "22": "glass_box",
    "23": "guitar",
    "24": "keyboard",
    "25": "lamp",
    "26": "laptop",
    "27": "mantel",
    "28": "person",
    "29": "piano",
    "30": "plant",

    "31": "radio",
    "32": "range_hood",
    "33": "sink",
    "34": "stairs",
    "35": "stool",
    "36": "tent",
    "37": "tv_stand",
    "38": "vase",
    "39": "wardrobe",
    "40": "xbox"
}
class_name_to_id = { v : k for k, v in class_id_to_name.items() }

class_names = set(class_id_to_name.values())
